package com.tianqi.tools;

import java.io.File;

import javax.sql.DataSource;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.druid.DruidPlugin;
import com.tianqi.config.JfinalConfig;

/**
 * 在数据库表有任何变动时，运行一下 main 方法，极速响应变化进行代码重构
 */
public class _JFinalDemoGenerator {
	
	public static DataSource getDataSource() {
		PropKit.use("a_little_config.txt");
		DruidPlugin druidPlugin = JfinalConfig.createDruidPlugin();
		druidPlugin.start();
		return druidPlugin.getDataSource();
	}
	
	public static void main(String[] args) {
		// model 所使用的包名 (MappingKit 默认使用的包名)
		// -------------------------------diy diy diy------------------------------------
		String modelPackageName = "com.tianqi.rms.model.business";
		
		 String path = PathKit.class.getResource("/").getFile();
		// base model 所使用的包名
		String baseModelPackageName = modelPackageName + ".base";
		// base model 文件保存路径F:\workspace\an_microPortal\WebContent----------
		System.out.println(path);
//		String baseModelOutputDir = PathKit.getWebRootPath() + "/../src/" + baseModelPackageName.replace(".", "/");
		String baseModelOutputDir = path + "../../src/" + baseModelPackageName.replace(".", "/");
		// model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
		String modelOutputDir = baseModelOutputDir + "/..";
		modelOutputDir = modelOutputDir.replace("/", File.separator);
		
		// 创建生成器
		Generator generator = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);
		DiyMappingKitGenerator diyMappingKitGen = new DiyMappingKitGenerator(modelPackageName, modelOutputDir);
		generator.setMappingKitGenerator(diyMappingKitGen);
		// 设置是否生成链式 setter 方法
		generator.setGenerateChainSetter(false);
		// 设置是否在 Model 中生成 dao 对象
		generator.setGenerateDaoInModel(true);
		// 设置是否生成字典文件
		generator.setGenerateDataDictionary(false);
		
		DiyMetaBuilder metaBuilder = new DiyMetaBuilder(getDataSource());
		// 添加需要生成的表名
		// -------------------------------diy diy diy------------------------------------
		metaBuilder.addIncludedTable("sms_send_info");
		generator.setMetaBuilder(metaBuilder);
		// 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
		generator.setRemovedTableNamePrefixes("sms_");	
		// 生成
		generator.generate();
	}
	
}




